package NO6_Tree;

/**
 * 2025-09-10
 * 力扣 - 129. 求根节点到叶节点数字之和
 * <p>
 * 给你一个二叉树的根节点 root ，树中每个节点都存放有一个 0 到 9 之间的数字。
 * 每条从根节点到叶节点的路径都代表一个数字：
 * 例如，从根节点到叶节点的路径 1 -> 2 -> 3 表示数字 123 。
 * 计算从根节点到叶节点生成的 所有数字之和 。
 * 叶节点 是指没有子节点的节点。
 */
public class SumNumbersMain {
    public static void main(String[] args) {
        TreeNode root = new TreeNode(1, new TreeNode(2), new TreeNode(3));
        System.out.println(sumNumbers(root));
    }

    public static int sumNumbers(TreeNode root) {
        return dfs(root, 0);
    }

    // 先序DFS遍历
    private static int dfs(TreeNode node, int currentSum) {
        if (node == null) {
            return 0;
        }

        // 计算当前路径的数字
        currentSum = currentSum * 10 + node.val;

        // 如果是叶子节点，返回当前数字
        if (node.left == null && node.right == null) {
            return currentSum;
        }

        // 递归计算左右子树
        int leftSum = dfs(node.left, currentSum);
        int rightSum = dfs(node.right, currentSum);

        return leftSum + rightSum;
    }
}
